import {
    Context,UserModel
} from 'hydrooj';


export async function apply(ctx:Context){   
    ctx.on("handler/after/ContestList#get",async (handler)=>{       
        try {
            // 确保 tdocs 存在且不为空
            if (handler.response.body?.tdocs && handler.response.body.tdocs.length > 0) {
                // 获取所有比赛创建者的用户ID
                const ownerIds = handler.response.body.tdocs.map(tdoc => tdoc.owner);
                // 执行你想要添加的代码
                const udict = await UserModel.getList(handler.args.domainId, ownerIds);
                // 将用户数据添加到返回结果中
                handler.response.body.udict = udict;
            }
        } catch (error) {
            handler.response.body.msg = error.message;
        }
    });
    ctx.on("handler/after/HomeworkMain#get",async (handler)=>{
        try{
            if (handler.response.body?.tdocs && handler.response.body.tdocs.length > 0) {
                // 获取所有比赛创建者的用户ID
                const ownerIds = handler.response.body.tdocs.map(tdoc => tdoc.owner);
                // 执行你想要添加的代码
                const udict = await UserModel.getList(handler.args.domainId, ownerIds);
                // 将用户数据添加到返回结果中
                handler.response.body.udict = udict;
            }
        }
        catch(error){
            handler.response.body.msg = error.message;
        }
    });
}

